package com.hmdp.utils;

import cn.hutool.core.util.BooleanUtil;
import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSON;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.hmdp.dto.Result;
import com.hmdp.entity.Shop;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.util.Objects;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.function.Function;

import static com.hmdp.utils.RedisConstants.*;

/**
 * 操作缓存的工具类
 */
@Slf4j
@Component
public class CacheClient {


    private static final ExecutorService CACHE_REBUILD_EXECUTOR = Executors.newFixedThreadPool(10);

    private StringRedisTemplate stringRedisTemplate;

    public CacheClient(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }

    /**
     * 将任意java对象序列化为json并存储在string类型的key中，并且可以设置过期时间
     */
    public void set(String key, Object value, Long time, TimeUnit timeUnit){

        stringRedisTemplate.opsForValue().set(key, JSONUtil.toJsonStr(value),time,timeUnit);
    }


    /**
     * 将任意Java对象序列化为json并存储在string类型的key中，
     *    并且可以设置逻辑过期时间，用于处理缓存击穿问题
     * @param key
     * @param value
     * @param time
     * @param timeUnit
     */
    public void setWithLogicalExpire(String key, Object value, Long time, TimeUnit timeUnit){
        RedisData redisData = new RedisData();
        redisData.setData(value);
        //设置逻辑过期时间
        redisData.setExpireTime(LocalDateTime.now().plusSeconds(timeUnit.toSeconds(time)));
        //写入redis
        stringRedisTemplate.opsForValue().set(key,JSONUtil.toJsonStr(redisData));
    }

    /**
     * 根据指定的key查询缓存，并反序列化为指定类型，
     * 利用缓存空值的方式解决缓存穿透问题
     * @param keyPrefix
     * @param id
     * @param type
     * @param dbFallback
     * @param time
     * @param unit
     * @return
     * @param <R>
     * @param <ID>
     */

    public <R,ID> R  queryWithPassThrough(String keyPrefix, ID id, Class<R> type,
                                          Function<ID,R> dbFallback,Long time,TimeUnit unit){
        String key = keyPrefix + id;
        //从redis中查询数据
        String json = stringRedisTemplate.opsForValue().get(key);
        //缓存命中直接返回店铺数据
        if(StrUtil.isNotBlank(json)){
            return JSONUtil.toBean(json,type);
        }
        //如果等于空值，即缓存的空对象，直接返回null
        if(json != null){
            return null;
        }
          //数据为null,则去数据库查
           R r = dbFallback.apply(id);

        if(r == null){
            stringRedisTemplate.opsForValue().set(key,"",CACHE_NULL_TTL,TimeUnit.SECONDS);
            return null;
        }
        //存在则写入redis
        this.set(key,r,time,unit);
        return r;

    }

    /**
     *利用逻辑过期时间解决缓存击穿问题
     * @param keyPrefix
     * @param id
     * @param type
     * @param dbFallback
     * @param time
     * @param unit
     * @return
     * @param <R>
     * @param <ID>
     */

    public <R,ID> R queryWithLogicalExpire(String keyPrefix, ID id ,Class<R> type,Function <ID,R> dbFallback,
                                           Long time, TimeUnit unit ){
        String key = keyPrefix + id;
        String json = stringRedisTemplate.opsForValue().get(key);

        if(StrUtil.isBlank(json)){
            return  null;
        }
        RedisData redisData = JSONUtil.toBean(json, RedisData.class);
        JSONObject data = (JSONObject)redisData.getData();
        R r = JSONUtil.toBean(data, type);
        LocalDateTime expireTime = redisData.getExpireTime();
        //判断是否逻辑过期
        if(expireTime.isAfter(LocalDateTime.now())) {
            //没过期直接返回店铺信息
            return r;
        }
        //已过期则进行缓存重建
        String lockKey = LOCK_SHOP_KEY + id;

        // 2、缓存数据已过期，获取互斥锁，并且重建缓存
        boolean isLock = tryLock(lockKey);
        if(isLock){
            // 获取锁成功，开启一个子线程去重建缓存
          CACHE_REBUILD_EXECUTOR.submit(()->{
            try{

                R newR = dbFallback.apply(id);
                // 将查询到的数据保存到Redis
                this.setWithLogicalExpire(key,newR,time,unit);

            }catch (Exception e){
                throw new RuntimeException(e);

            }finally {
                //释放锁
                unLock(lockKey);
            }
          });
        }
        // 3、获取锁失败，再次查询缓存，判断缓存是否重建（这里双检是有必要的）
          json = stringRedisTemplate.opsForValue().get(key);
        if(StrUtil.isBlank(json)){
            // 3.1 缓存未命中，直接返回失败信息
            return  null;
        }
        //  3.2 缓存命中，将JSON字符串反序列化未对象，并判断缓存数据是否逻辑过期
      redisData = JSONUtil.toBean(json,RedisData.class);
      expireTime = redisData.getExpireTime();
        // 这里需要先转成JSONObject再转成反序列化，否则可能无法正确映射Shop的字段
        data =(JSONObject)redisData.getData();
        JSONUtil.toBean(data,type);
        if(expireTime.isAfter(LocalDateTime.now())){
          //当前数据未过期，直接返回
          return r;
      }
      //4.返回过期数据
      return r;
    }


    /**
     * 利用互斥锁解决缓存击穿问题
     * @param keyPrefix
     * @param id
     * @param type
     * @param dbFallback
     * @param time
     * @param unit
     * @return
     * @param <R>
     * @param <ID>
     */

    public <R,ID> R queryWithMutex(String keyPrefix,ID id,Class<R> type,
                                   Function<ID,R> dbFallback,Long time,TimeUnit unit) {

        String key = keyPrefix + id;
        //1.从redis中查询店铺数据，判断缓存是否命中
        String json = stringRedisTemplate.opsForValue().get(key);
        //2.缓存命中则直接返回,即有值存在.
        if (StrUtil.isNotBlank(json)) {
            return JSONUtil.toBean(json, type);
        }
        //判断是否是空值，是就返回null
        if (json != null) {
            return null;
        }
        //3.缓存未命中,需要重建缓存，判断是否能够获取互斥锁
        String lockKey = LOCK_SHOP_KEY + id;
        boolean flag = tryLock(lockKey);
        R r = null ;
        //4.未获得锁，说明已有线程在重建缓存,先休眠然后在重新去查询缓存
        try {
            if (!flag) {
                //获得失败，失眠并重试
                Thread.sleep(50);
                return queryWithMutex(key, id, type, dbFallback, time, unit);
            }
            //5.获得锁,先判断缓存是否要重建，防止堆积的线程全部访问数据库
            json = stringRedisTemplate.opsForValue().get(key);

            //不为空则直接返回
            if (StrUtil.isNotBlank(json)) {
                return JSONUtil.toBean(json, type);
            }

            if (json != null) {

                return null;
            }
            //7.未命中,则去数据库查对应的商户信息
             r = dbFallback.apply(id);

            if (r == null) {
                this.set(key, "", CACHE_NULL_TTL, TimeUnit.SECONDS);
                return null;
            }
            //存在写入redis
            this.set(key, r, time, unit);


        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            unLock(lockKey);
        }
        return r;
    }



    private boolean tryLock(String key) {

        Boolean flag = stringRedisTemplate.opsForValue().setIfAbsent(key, "1", LOCK_SHOP_TTL, TimeUnit.SECONDS);
        return BooleanUtil.isTrue(flag);
    }

    private void unLock(String key) {
        Boolean delete = stringRedisTemplate.delete(key);
    }

}
